package com.hup.utils.algorithm.image;

import java.awt.*;
import java.awt.geom.Area;
import java.awt.image.PixelGrabber;
import java.util.ArrayList;

/**
 * 获取图像外轮廓
 * 搜索中文注释即找到很多不一样的文章,但代码是一样的
 * https://blog.csdn.net/iteye_7550/article/details/82238286
 *
 * @author hugan
 * @date 2020/4/14
 */
public class ImageShape {

    /**
     * 将Image图像转换为Shape图形
     *
     * @return Image图像的Shape图形表示
     */
    public static Shape getImageShape(Image img) {
        ArrayList<Integer> x = new ArrayList<>();
        ArrayList<Integer> y = new ArrayList<>();
        int width = img.getWidth(null);//图像宽度
        int height = img.getHeight(null);//图像高度

        //筛选像素
        //首先获取图像所有的像素信息
        PixelGrabber pgr = new PixelGrabber(img, 0, 0, -1, -1, true);
        try {
            pgr.grabPixels();
        } catch (InterruptedException e) {
            throw new IllegalArgumentException("图像解析异常?:" + e.getMessage());
        }
        int pixels[] = (int[]) pgr.getPixels();

        //循环像素
        for (int i = 0; i < pixels.length; i++) {
            //筛选，将不透明的像素的坐标加入到坐标ArrayList x和y中
            int alpha = getAlpha(pixels[i]);
            if (alpha != 0) {
                x.add(i % width > 0 ? i % width - 1 : 0);
                y.add(i % width == 0 ? (i == 0 ? 0 : i / width - 1) : i / width);
            }
        }

        //建立图像矩阵并初始化(0为透明,1为不透明)
        int[][] matrix = new int[height][width];
        for (int i = 0; i < height; i++) {
            for (int j = 0; j < width; j++) {
                matrix[i][j] = 0;
            }
        }

        //导入坐标ArrayList中的不透明坐标信息
        for (int c = 0; c < x.size(); c++) {
            matrix[y.get(c)][x.get(c)] = 1;
        }

             /* 由于Area类所表示区域可以进行合并，我们逐一水平"扫描"图像矩阵的每一行，
             * 将不透明的像素生成为Rectangle，再将每一行的Rectangle通过Area类的rec
              * 对象进行合并，最后形成一个完整的Shape图形
             */
        Area rec = new Area();
        int temp = 0;

        for (int i = 0; i < height; i++) {
            for (int j = 0; j < width; j++) {
                if (matrix[i][j] == 1) {
                    if (temp == 0) {
                        temp = j;
                    }
                } else {
                    if (temp != 0) {
                        Rectangle rectTemp = new Rectangle(temp, i, j - temp, 1);
                        rec.add(new Area(rectTemp));
                        temp = 0;
                    }
                }
            }
            temp = 0;
        }
        return rec;
    }

    /**
     * 获取像素的Alpha值
     *
     * @return 像素的Alpha值
     */
    private static int getAlpha(int pixel) {
        return (pixel >> 24) & 0xff;
    }

}
